* Code improvement + tests * review * review + moving to pytest test function * move test to async * remove code comment