Fixed prime bug, completed assignment #4