Adding check for completion block