Class assignment 2