Add Feature Image and polished code