Input dimension error fixed...