Add generator function / remove unnecessary whitespace