Minor fixups

Missing import; spacing fix
Making better use of CoffeeScript's pattern matching