Adding pk_prototype to calculate Power Spectrum