This should be moved after running graphlab :)