Tweeks to assignment 3