Small edit to logic and whitespace

This is where I changed the logic.