Code for Problems 1 and 2 added