More code solution