I think it's correctly work, but I have problems with testing