Assignment on Math Problem