Nicer code
Added prime sum implementation