Modifications to calculate power