Implement Mlp From Scratch Python