Clean test/main.py
Update Readme
Add some docstring