Add real tests
Add real tests