update_flags->raw = 0; // Reset all flags
 
+       if (u->flip_addr)
+               update_flags->bits.addr_update = 1;
+
        if (!is_surface_in_context(context, u->surface)) {
                update_flags->bits.new_plane = 1;
                return UPDATE_TYPE_FULL;
 
 }
 
 
-static void dcn20_disable_plane(struct dc *dc, struct pipe_ctx *pipe_ctx)
+void dcn20_disable_plane(struct dc *dc, struct pipe_ctx *pipe_ctx)
 {
        DC_LOGGER_INIT(dc->ctx->logger);
 
 
        dcn20_plane_atomic_disable(dc, pipe_ctx);
 
-       //apply_DEGVIDCN10_253_wa(dc);
-
        DC_LOG_DC("Power down front end %d\n",
                                        pipe_ctx->pipe_idx);
 }
        }
 }
 
-static void dcn20_enable_plane(
+void dcn20_enable_plane(
        struct dc *dc,
        struct pipe_ctx *pipe_ctx,
        struct dc_state *context)
 }
 
 
-void dcn20_program_pipe(
+static void dcn20_program_pipe(
                struct dc *dc,
                struct pipe_ctx *pipe_ctx,
                struct dc_state *context)
 
                struct pipe_ctx *pipe,
                bool lock);
 void dcn20_setup_gsl_group_as_lock(const struct dc *dc,
-                                 struct pipe_ctx *pipe_ctx,
-                                 bool enable);
-
+                               struct pipe_ctx *pipe_ctx,
+                               bool enable);
+void dcn20_pipe_control_lock(
+       struct dc *dc,
+       struct pipe_ctx *pipe,
+       bool lock);
+void dcn20_disable_plane(struct dc *dc, struct pipe_ctx *pipe_ctx);
+void dcn20_enable_plane(
+       struct dc *dc,
+       struct pipe_ctx *pipe_ctx,
+       struct dc_state *context);
 #endif /* __DC_HWSS_DCN20_H__ */