Adding graph back to the main screen