Reverted the changes to the main script