Updated Graph Class and Tests