Code changes to make it run on my computer