Added JUnit tests

Changed wrong return type in StageManager