Fixed a bug and updated to 0.2