added more code

main.jave