assignment 1 - fix problem 1 notations of formula