Example of unit testing
Format code following pep8