Forgot to change to master branch. Started adding func to add attributes