Small changes so the program runs