added more controller functiona;ity