Change the keylogger class and write a better script