Improve key bindings for align