Finished designing the UI in section 4