Modify script so that it can be used at work and home