adding tests - bit broken