Add validation and helpful messages to the UI