modified main.java
demo