Added final comments/UI stuff

Final before submission