Moved out atom color assignment