Need to work on the saving/loading functions