Code file for the problem