Add partials, refactor test_spec.rb file