add a print and rework if elif else condition