Updated module to be more Python friendly.

Thanks Whoopska for the pointers. :)