Fixed minor formatting bugs in UI and Code Output