change original class behavior