Tried adding tests