Fixed prime function for numbers less than 2