Code that handles the final_design. Still needs work!