improve sample code section