Phone using mobile data instead of Wi-Fi
Go into the Wi-Fi settings, press Menu, choose Advanced. You should get options including Wi-Fi Sleep Policy. Sounds like its set to turn off Wi-Fi when the screen is off.