writing unit tests and adding nose