I'm not sure if my approach is correct