Small changes to push script