forgot a part of code