Code works. Tests written