changes to Junit controller