putting dropout back in - doesn't explain bug with hidden nodes