added more test to the controller