forgot 2 lines of code