add main()

Added a main method which allows users to interact with the program