code of problem A