Added tests for the main controller