adding pytest to dependencies
removing unused code