Added some contract tests