题目给的代码不怎么严谨，请问怎么修改更好

这段代码主要讲的是关于String的相关用法，当然也有一些我自己的看法：

一开始创建的fileName，假如这一段字符串没有"."  

那么之后的int index = fileName.lastIndexOf(".");中，index的值为 -1

继而之后的String prefix = fileName.substring(index);就会报错

我自己把这段代码这么修改了一下：

if(index != -1){

String prefix = fileName.substring(index);

if (index>0 && prefix.equals(".java")) {

System.out.println("Java文件名正确");

} 

else {System.out.println("Java文件名无效");

}

}else{System.out.println("Java文件名无效");

}

请问还有什么更好的修改方法