Add unit tests, Add DB schema