Clean up code
Update version 0.1.0