Fix small bug and add more to README