Clean up code
Tagging 0.1.0