Removed useless lines on controller + applied class User