cleaning up code. updating readme.