For assignment 6