Remove unnessicary whitespace
Update example to refactored code