Updated implementation example