Forgot to add the Edge class.