minor changes so it can run