Update prog02

Made it sooooo much better
uncomment the stuff in main to see
what I'm talking about