Made a few touch ups to the code