Working more on the code, but still not working.