Adding more code on java