Improve database module