Cleaned up code and made a better README