Added tests for #4
Cleaned up some spaces in the code
Updated to version 0.3