Updated

added controller functionality