Parameterised the number of labels
Moved belief propagation code into it's own files, modularised code