Feature/use pytest