diff --git a/app/controllers/sources_controller.rb b/app/controllers/sources_controller.rb
--- a/app/controllers/sources_controller.rb
+++ b/app/controllers/sources_controller.rb
@@ -1,7 +1,17 @@
 class SourcesController < ApplicationController
-  before_filter :authenticate, :except => [:index, :login]
+  before_filter :authenticate
 
   def direct_edit
-    @problem = Problem.find_by_id(params[:pid])
+    @problem = Problem.find(params[:pid])
+    @source = ''
   end
+
+  def direct_edit_submission
+    @submission = Submission.find(params[:sid])
+    @source = @submission.source.to_s
+    @problem = @submission.problem
+    @lang_id = @submission.language.id
+    render 'direct_edit'
+  end
+
 end