Fixing an import
Fixing some sample code errors