Add code to solution