Cleaned up code and readme 2