Neural network in three lines of code