Re-working #18 and adding documentation. Also bumping version to 0.2.0