work on if without else as well.