added some tests
added a blank line to be compliant with python PEP