Small changes

Still trying to fix problem