Added save function
Can't remember.