code change

made code a bit shorter