もちろん
```
import math
def speed_after_impact(gravity, height):
    return math.sqrt(2*gravity*height)
```
Sure!
```
import math
def speed_after_impact(gravity, height):
    return math.sqrt(2*gravity*height)
```