Code tidy
0.1.1