Add in code for SA

Need to revise a bit