Add upgrade for database