Finishing up sample code