Cleaned the code even more, added headers. Enjoy reading beautiful code!!