Add type in controller and some css