Cleaned up code, wrote README