Made it look prettier

Still need to do a menu option from the beginning and add error
handling. I hate error handling.