changed the output size from 3 to 2