Updated module for Prezto

(Should) work with both Prezto and OMZ

[ticket: X]