For assignment 2