Add condition for numbers less than 2 in method prime?