Add puma and rbx support