cleaning up the readme and code a bit