Shortened code a litte.