Add schemas code

You don't need to import classes that are in the same package.