Almost there. need to solve load/save file issue