added gradient back