Minor Update v0.1.1

Prettify Code