shorter code

Make code shorter