Add S4 code