Controller tests complete, fixed tree drawing bug

controller tests added. forgot to remove tree drawing code for same
value nodes which caused some nodes not to be coloured. fixed this