Initial solution for Q2. Changed == to .equals()