init
* catching * fix * improve * fix linter * cleanup * fix prediction * do the same here * fix comment