Fix #1: refactor with rubocop