Fixed a null pointer exception in Main