Added button check