Add return_epg function