Finished basic assignment. Trying to do extra credit