Bumped version to 0.0.2 and removed TODO