improve big_oh automation a little