Editing my code after holidays