Refactoring

Add DB connection module
Add helper for test