fixed a couple of little things before submitting a PR