Removed unnecessary imports, added another example