fixes #3 added :)