actually added the changes + a test .py file