Solved the first part but another problem has appeared