First draft completed. Still needs some refinement to the UI