Added some validation. Should move into a text file or something.