Style improvements

Done along with @mollymcleod!
Was this removal intentional?