Add return for chainability