renamed class Main to Controller