In middle of implementing button functionality