0.1.1
cleaned up code