Added brush menu and brush switch logic