Small change to compile and run