Removed import for generate_pbs function