ENH: combine norm functions, take axis argument