Cleaning up code a bit
Adding README