Cleanup
Use log instead of print