replace print statement with print function