Adding in os.listdir()
More work on code upload