fixed linter errors, need to add curly bracket to pass tests

fixed linter errors, need to add curly bracket to pass tests #21