Updated Code

added more UI elements