Add code for create db