Cleaned up more code, updated docstrings & updated readme.