Changed to if loop, fixed spacing on print statement