Added controls functionality