Add registration test
Add proxy model tests