Added prime generation