Made some changes to the ex2