Add a wrapper for rails