Adding btn functionality