Added functionality to create button