Fixing print in main