Updated to assignment 2.