Add docs and comply with Rubocop