Adding rails 3.1.3 support