Updating version to 0.1 and a few small changes