EVERYTHING WORKS FOR ASSIGNMENT 2 :)