Exercise1: add forgotten part