Make easier to run by using python get directory names