Add relevant test