modified main cpp

This reverts commit c9be702a59863b367586ed4384274ae734edb807.