Input and slight modifications so it compiles on my machine