keep_units function

Apply a function keeping units