Functions again

CodeForces - 788A