fixes #1

Small changes...