Fixed minor bug on the prototype UI. Need to refine it still.