Improved the styling somewhat, added an error msg