Mock database and class add done!