Remove print statement at end of code