Co-authored-by: Milena Zlaticanin <60530402+Zlaticanin@users.noreply.github.com>
* adding copyright header * fix fmt and a test