More unit tests added graph