Have to complete this program. Still not working properly