Code improvements

Set the version to 1.0.1
Made the drawing code more efficient