I've added some code to the question which gives an idea of the problem.

For some reason the code wasn't being displayed with formatting when I did an update but it appears to be working now.