Finished section 5.11 - Adding some validation