add db funcitonality