Cleaned up code a little and added link to my github page