I use a "tomato" pin cushion that I marked in sections with a sharpie, put an identifier as to what size/type needle in each section and then stick my needles in it as I remove them from the machine.
Edit to add: I'm sorry, I didn't see this had already been suggested.