Revert "Tried to add dropout"

This reverts commit 78ebd258a93c938ddde63bdf12bcfa37ff970489.