Add puzzles to contests

app/controllers/contests_controller.rb

@@ -8,6 +8,7 @@ class ContestsController < ApplicationController
 
   def show
     @title = @contest.name
+    @puzzles = @contest.puzzles
     set_badges
   end
 

app/controllers/puzzles_controller.rb

@@ -1,4 +1,5 @@
 class PuzzlesController < ApplicationController
+  before_action :set_contest
   before_action :set_puzzle, only: %i[ show destroy ]
 
   def index
@@ -10,13 +11,16 @@ class PuzzlesController < ApplicationController
 
   def new
     @puzzle = Puzzle.new
+    @title = "New contest puzzle"
   end
 
   def create
     @puzzle = Puzzle.new(puzzle_params)
+    @puzzle.contest_id = @contest.id
     if @puzzle.save
-      redirect_to @puzzle
+      redirect_to contest_path(@contest)
     else
+      @title = "New contest puzzle"
       render :new, status: :unprocessable_entity
     end
   end
@@ -28,6 +32,10 @@ class PuzzlesController < ApplicationController
 
   private
 
+  def set_contest
+    @contest = Contest.find(params[:contest_id])
+  end
+
   def set_puzzle
     @puzzle = Puzzle.find(params[:id])
   end