Updated simple key driver

Simplified usage of key checking