nicer code
remove unused imports
use also postgres database