.cs file. Make changes to these two functions

// ADD is the stuff that needs to be added