If you find any better way to implement functions like forward propagation, activation functions,…etc. Please inform