Header changes

Move header back to controller
Header changes

Move header back to controller
Merge pull request #9 from Dimasdanz/dimas

Header changes looks good