Added information on how to run the code.
Cleaned up the code a little.