Glad you like it! The game actually tells you how to handle this: use the Mouse Wheel to switch modes, as shown on the screen in the hallway. When you try to take the color of an object the text box at the bottom will also remind you to change your mod. There is also indication on the modules themselves when you pick them up.