Added both files, completed

For some reason, as I was coding, Github wasn't updating and allowing
me to commit changes I made. So this is the completed code.