Made the custom controller to look pretty