Added a main function. Old habits die hard