updated UI and got graph working