Adds tests for main controller