Added kMag function to compute k

1. Given energy and angle