Just a small change to make it more general. Maybe you can use that :)