Minor changes to noob solution