Small changes to initial repo