Add mouse checking

- Replace key check with mouse check