Fixed module importing and added module documentation