I don't like xpath since css selector is comparingly faster and preferrable option. (So, this is just a suggestion)
Maybe change it to use css selector is better :P