Move controller method into controller class
Merge pull request #9 from asasse01/refactor_main_class

Move controller method into controller class