Functionality done. Needs more UI changes.