comsetic corrections - now it is functional :)