Fixing A Main Function