add code and setup.py
format readme