small changes with button