Adding code from assignment 3