Complete Work on Assignment UI