Changed == to .compareTo for better string comparison