I added this aswell :), see rebased version.