Fixed style, tests and docs for first problem