Added Discussion section and made improvements to UI