Add strong params and error message styling