change to method so not hardcoded