changes to main script