Added keypress and finished rainbow