Solution for my part of group challenge