Added package declaration and fixed indentation