Adding in main script