Removed documentation and renamed in preparation to add dropout