Got key checking working