Cleaned up the code a little and added some more comments to the info section