Added some basic stuff to the main function...