Fixed rubocop errors and remove unittest