Removing prints from code