Modifying to make 'z' key work