Update/cleanup, changed constructor

Signed-off-by: Przemyslaw Springer <przmk01@gmail.com>
Minor update

Signed-off-by: Przemyslaw Springer <przmk01@gmail.com>