Added import statement to sample code