Update setOption to return the current class