Modify the javaFX code in ch04