Create function to turn all nonzero elements into 1's