Minor bug fixing adding button test cases