add after_push functionality