fixing code, adding linear rectifier