Some whitespace breaking up the assignments and if statements would be nice here