Fixing HW6 to work with linux and mac