Added button controller