Same swap but here we use functions