Add screen wrapping for enemies