Clean up code
0.1.7