new edit to fix img problems