improve code readability in Assignment-0