Should add protocol here