Rewrite 'setup' function
Add db schema