fixed problem with main program
fixed problem with main program