Cleaned up some syntax with rubocop