Bug FIxes for 0.1.0