Bug fix.. trying to add key listener