Fix some bugs. Change to rectifier linear