Clean up sample code