Add option to change node size