cleaned up a little code, and added to the readme