Updated to 1.0.1 and fixed a couple of minor bugs