Add part of logic