Optimize the sample code;