Added part of solution