added some tests to controller