Fixes example and removed some unused use statements