Added setup for database