Simplifying problem section and adding solution section.