added assignment 2 a and b