Got power spectrum working